# tests/test_missing_split_export.py
import os
import pandas as pd
from missing_split_export import export_missing_data_by_indicator

def test_export_missing_data_by_indicator(tmp_path):
    # 使用pytest的tmp_path临时目录
    data = pd.DataFrame({
        '国名Ch': ['中国', '美国', '日本'],
        'Year': [2020, 2020, 2020],
        'GDP': [14.7, None, 5.0],
        '人口': [1400, 328, None]
    })
    
    output_dir = tmp_path / "output"
    export_missing_data_by_indicator(data, output_dir)
    
    # 检查文件是否生成
    assert (output_dir / "GDP.csv").exists()
    assert (output_dir / "人口.csv").exists()
    
    # 检查内容是否正确
    gdp_data = pd.read_csv(output_dir / "GDP.csv")
    assert len(gdp_data) == 1  # 只有美国有缺失
    assert gdp_data.iloc[0]['国名Ch'] == '美国'